package org.skywork.top.core.algorithm.num;

/**
 * 快乐数
 */
public class no_202 {


    /**
     * 编写一个算法来判断一个数 n 是不是快乐数。
     *
     * 「快乐数」 定义为：
     *
     * 对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
     * 然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。
     * 如果这个过程 结果为 1，那么这个数就是快乐数。
     * 如果 n 是 快乐数 就返回 true ；不是，则返回 false 。
     */
    public boolean isHappy(int n) {

        /**
         * 示例 1：
         * 输入：n = 19
         * 输出：true
         * 解释：
         * 1^2 + 9^2 = 82
         * 8^2 + 2^2 = 68
         * 6^2 + 8^2 = 100
         * 1^2 + 0^2 + 0^2 = 1
         *
         *
         * 示例 2：
         * 输入：n = 2
         * 输出：false
         */

        return isHappy2(String.valueOf(n));
    }

    public boolean isHappy2(String n) {

        //1. 判断是否只有一个数, 不是1的直接返回false
        char[] chars = n.toCharArray();
        if(chars.length == 1){
            if(chars[0] == '1'){
                return true;
            }else{
                return false;
            }
        }
        int sum = 0;
        for (char aChar : chars) {
            int pow = (int) Math.pow((double) aChar, 2);
            sum += pow;
        }
        if(sum % 5 == 0){
            return true;
        }
        return isHappy2(String.valueOf(sum));
    }

    public static void main(String[] args) {
        no_202 no_202 = new no_202();
        boolean flag = no_202.isHappy(1);
        System.out.println(flag);
    }
}
